package com.demo.jzoffer2;

import org.junit.Test;

/**
 * @author gy
 * @date 2023/03
 */
public class Test0326_01 {

    /**
     * 地上有一个m行n列的方格，从坐标 [0,0] 到坐标 [m-1,n-1] 。一个机器人从坐标 [0, 0] 的格子开始移动，
     * 它每次可以向左、右、上、下移动一格（不能移动到方格外），也不能进入行坐标和列坐标的数位之和大于k的格子。
     * 例如，当k为18时，机器人能够进入方格 [35, 37] ，因为3+5+3+7=18。但它不能进入方格 [35, 38]，因为3+5+3+8=19。
     * 请问该机器人能够到达多少个格子？
     *
     * 示例 1：
     *
     * 输入：m = 2, n = 3, k = 1
     * 输出：3
     *
     * 示例 2：
     *
     * 输入：m = 3, n = 1, k = 0
     * 输出：1
     *
     * 提示：
     *
     * 1 <= n,m <= 100
     * 0 <= k <= 20
     * @param m
     * @param n
     * @param k
     * @return
     */
    public int movingCount(int m, int n, int k) {
        boolean[][] visited = new boolean[m][n];
        return dfs(visited, m, n, k, 0, 0);
    }

    /**
     * 函数定义：获取本单元格递归搜索的可达解总数
     * 深度优先搜索：可以理解为暴力发模拟机器人在矩阵中的所有路径。DFS通过递归，先朝一个方向搜到底
     * 再回溯至上个节点，沿另一个方向搜索
     * 剪枝：在搜索中，遇到数位和超过目标值，此元素已经访问，则应该立即返回，称之为可行性剪枝
     * @param visited 可达元素
     * @param m 行数
     * @param n 列数
     * @param k 目标值
     * @param i 行索引
     * @param j 列索引
     * @return
     */
    private int dfs(boolean[][] visited, int m, int n, int k, int i, int j) {
        // 寻找终止条件：1、当行列索引越界时 2、数位和超过目标值 k 3、当前元素已经访问过时
        if(i >= m || j >= n || visited[i][j] || bitSum(i) + bitSum(j) > k) {
            return 0;
        }
        // 标识此处可达
        visited[i][j] = true;
        // 找到函数的等价关系式：本单元格递归搜索的可达解总数 = 1(开始位置) + 右方搜索的可达解总数 + 下方搜索的可达解总数
        return 1 + dfs(visited, m, n, k, i + 1, j) + dfs(visited, m, n, k, i, j + 1) ;
    }

    /**
     * 1 <= n,m <= 100
     * 0 <= k <= 20
     * 计算某个数字的 数位和
     * eg：34 -> 3 + 4 = 7
     * @param n
     * @return
     */
    private int bitSum(int n) {
        int sum = 0;
        while(n > 0) {
            // 获取最低位数字
            sum += n % 10;
            // 获取最低位的上一位的数字
            n /= 10;
        }
        return sum;
    }

    @Test
    public void m1() {

    }

    @Test
    public void m2() {
        }

    @Test
    public void m3() {

    }



}
